// Muduo - A reactor-based C++ network library for Linux
// Copyright (c) 2010, Shuo Chen.  All rights reserved.
// http://code.google.com/p/muduo/
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//
//   * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//   * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//   * Neither the name of Shuo Chen nor the names of other contributors
// may be used to endorse or promote products derived from this software
// without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


#ifndef CLUSNET_BASE_THREADPOOL_H
#define CLUSNET_BASE_THREADPOOL_H

#include <clusnet/base/Thread.h>
#include <clusnet/base/Mutex.h>

#include <condition_variable>
#include <functional>
#include <memory>
#include <deque>
#include <vector>
#include <string>

namespace clusnet
{
namespace base
{

class ThreadPool final
{
    public:
        using Task = std::function<void ()>;

        ThreadPool(const ThreadPool&) = delete;
        ThreadPool& operator= (const ThreadPool&) = delete;
        explicit ThreadPool(size_t maxSize, const std::string& nameArg = std::string("ThreadPool"), const Task& cb = Task());
        ~ThreadPool();

    public:
        void start(int numThreads);
        void stop();

        const std::string& name() const
        {
            return name_;
        }

        size_t queueSize() const;

        // Could block if maxQueueSize > 0
        void run(Task&& f);
        void run(const Task& f);

    private:
        bool isFull() const;
        void runInThread();
        Task take();

    private:
        mutable Mutex       mutex_;
        std::condition_variable_any notEmpty_;
        std::condition_variable_any notFull_;
        std::string         name_;
        Task                threadInitCallback_;
        std::vector<std::shared_ptr<clusnet::base::Thread>>   threads_;
        std::deque<Task>    queue_;
        size_t              maxQueueSize_;
        bool                running_;
};
}
}


#endif 
